package cn.wolfcode.wolf2w.service.impl;

import cn.wolfcode.wolf2w.domain.StrategyTheme;
import cn.wolfcode.wolf2w.query.StrategyThemeQuery;
import cn.wolfcode.wolf2w.repository.StrategyThemeRepository;
import cn.wolfcode.wolf2w.service.IStrategyThemeService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.*;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;

import java.util.List;

@Service
public class StrategyThemeServiceImpl implements IStrategyThemeService {

    @Autowired
    private StrategyThemeRepository strategyThemeRepository;

    @Autowired
    private MongoTemplate mongoTemplate;
    @Override
    public void save(StrategyTheme strategy) {
        strategy.setId(null);
        strategyThemeRepository.save(strategy);
    }

    @Override
    public void update(StrategyTheme strategy) {
         strategyThemeRepository.save(strategy);
    }

    @Override
    public void delete(String id) {
    strategyThemeRepository.deleteById(id);
    }

    @Override
    public StrategyTheme get(String id) {
        return strategyThemeRepository.findById(id).get();
    }

    @Override
    public List<StrategyTheme> list() {
        return strategyThemeRepository.findAll();
    }


    @Override
    public Page<StrategyTheme> query(StrategyThemeQuery qo) {
        //查询对象
        Query query = new Query();

        Long totalCount = mongoTemplate.count(query, StrategyTheme.class);

        Pageable pageable = PageRequest.of(qo.getCurrentPage()-1,qo.getPageSize(), Sort.Direction.ASC,"_id");

        query.with(pageable);

        List<StrategyTheme> strategyThemes = mongoTemplate.find(query, StrategyTheme.class);
        return new PageImpl(strategyThemes,pageable,totalCount);
    }

    @Override
    public void saveOrUpdate(StrategyTheme strategyTheme) {
        if(StringUtils.hasLength(strategyTheme.getId())){
            this.update(strategyTheme);
        }else {
            this.save(strategyTheme);
        }
    }


}
